'''
Author: cw
Date: 2025-08-18
Description: ...
'''

from modules.image_base_provider import ImageBaseProvider
from utils.local_server import client


class ImageLocalProvider(ImageBaseProvider):
    """本地 生成图片"""
    async def create(self, prompt: str):
        '''status'''
        return client.run(prompt)

    async def info(self, task_id):
        pass

    async def run(self, prompt: str, **_):
        """run"""
        return await self.create(prompt)